import pymysql

# 创建数据库连接对象
db = pymysql.connect(host="cdb-3q3rt2iy.cd.tencentcdb.com",port=10175,user="root",password="990331hz",database="bonus")


# 处理sql语句
def connect_mysql(sql,data):
    # 创建游标对象
    cursor = db.cursor()
    try:
        result = None
        # 如果data不为空
        if data:
            # 如果data是个列表对象，即data不止一条数据
            if isinstance(data[0],list):
                # 调用excetemany方法，将data中的数据都送入sql语句后再执行
                cursor.executemany(sql,data)
            else:
                # 否则直接调用execute方法，直接将data中都数据送入sql语句
                cursor.execute(sql,data)
        else:
            cursor.execute(sql)
            result = cursor.fetchall()
    except Exception as e:
        print(e)
        db.rollback()
    finally:
        cursor.close()
        db.commit()
        return result
